package com.qianfeng.iot.filters;

//
//                            _ooOoo_  
//                           o8888888o  
//                           88" . "88  
//                           (| -_- |)  
//                            O\ = /O  
//                        ____/`---'\____  
//                      .   ' \\| |// `.  
//                       / \\||| : |||// \  
//                     / _||||| -:- |||||- \  
//                       | | \\\ - /// | |  
//                     | \_| ''\---/'' | |  
//                      \ .-\__ `-` ___/-. /  
//                   ___`. .' /--.--\ `. . __  
//                ."" '< `.___\_<|>_/___.' >'"".  
//               | | : `- \`.;`\ _ /`;.`/ - ` : | |  
//                 \ \ `-. \_ __\ /__ _/ .-` / /  
//         ======`-.____`-.___\_____/___.-`____.-'======  
//                            `=---='  
//  
//         .............................................  
//                  佛祖镇楼                  BUG辟易  
//          佛曰:  
//                  写字楼里写字间，写字间里程序员；  
//                  程序人员写程序，又拿程序换酒钱。  
//                  酒醒只在网上坐，酒醉还来网下眠；  
//                  酒醉酒醒日复日，网上网下年复年。  
//                  但愿老死电脑间，不愿鞠躬老板前；  
//                  奔驰宝马贵者趣，公交自行程序员。  
//                  别人笑我忒疯癫，我笑自己命太贱；  
//  


import com.dc3.common.bean.R;
import com.dc3.common.constant.CacheConstant;
import com.dc3.common.constant.ServiceConstant;
import com.dc3.common.utils.Dc3Util;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.qianfeng.iot.openfeign.CacheFeignClient;
import io.jsonwebtoken.Claims;
import io.jsonwebtoken.Jwts;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.io.buffer.DefaultDataBufferFactory;
import org.springframework.stereotype.Component;
import org.springframework.util.AntPathMatcher;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Mono;

import java.io.IOException;
import java.nio.charset.StandardCharsets;
import java.util.Map;

/**
 * Created by Jackiechan on 2022/8/8 20:24
 *
 * @author Jackiechan
 * @version 1.0
 * @since 1.0
 */
@Component
public class TokenFilter implements GlobalFilter, Ordered {
    private ObjectMapper objectMapper;

    private CacheFeignClient cacheFeignClient;

    private AntPathMatcher antPathMatcher;

    @Autowired
    public void setAntPathMatcher(AntPathMatcher antPathMatcher) {
        this.antPathMatcher = antPathMatcher;
    }

    @Autowired
    public void setCacheFeignClient(CacheFeignClient cacheFeignClient) {
        this.cacheFeignClient = cacheFeignClient;
    }

    @Autowired
    public void setObjectMapper(ObjectMapper objectMapper) {
        this.objectMapper = objectMapper;
    }

    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        //检查令牌,我们可以在这里获取到令牌,然后进行检查
        //我们需要获取到用户的令牌数据
        String jwtHeaderValue = exchange.getRequest().getHeaders().getFirst(ServiceConstant.Header.X_AUTH_TOKEN);
        String requestPath = exchange.getRequest().getPath().toString();
        //如果是登录地址,就直接放行,登录是不需要检查令牌的
        if (antPathMatcher.match("/token/token", requestPath)) {
            return chain.filter(exchange);
        }
        //如果没有就不执行
        //通过解析令牌数据,得到里面的用户名+租户id
        try {
            return Mono.just(jwtHeaderValue)
                    .map(jwt -> jwt.split("\\."))//按照.进行分割
                    .map(strings -> strings[1])//获取到下标为1的数据,也就是存放数据的那一部分
                    .doOnError(throwable -> throwable.printStackTrace())//发生错误的时候会执行,我看看什么错误
                    .map(body -> Dc3Util.decode(body))//将正文的base64转成原始的json
                    .map(json -> {
                        try {
                            return objectMapper.readValue(json, Map.class);//转成map
                        } catch (IOException e) {
                            throw new RuntimeException(e);
                        }
                    }).flatMap(map ->
                    {
                        String sub = (String) map.get("sub");//获取到我们的用户名&租户id
                        String[] userNameTenantId = sub.split("&");//切成数组
                        //组成我们当前数据在redis中的key
                        String key = userNameTenantId[1] + CacheConstant.Suffix.TENANT_ID + CacheConstant.Entity.USER + userNameTenantId[0] + CacheConstant.Suffix.SALT;
                        return cacheFeignClient.getValue(key);//从redis中获取到盐
                    }).map(slat -> {
                        try {
                            Claims body = Jwts.parser().setSigningKey(slat.getBytes(StandardCharsets.UTF_8)).parseClaimsJws(jwtHeaderValue).getBody();
                            String subject = body.getSubject();//获取到我们的用户名&租户id
                            String tenantId = subject.split("&")[1];//获取到租户的id
                           //因为exchange是不可变的对象,里面的请求对象也是不可变,我们不能直接向里面的header添加数据,所以我们复制一个exchange出来
                            ServerWebExchange exchange1 = exchange.mutate()
                                    .request(builder -> {//修改request相关的数据
                                        builder.headers(httpHeaders -> {//我们修改一下请求头
                                            //添加了一个请求头
                                            httpHeaders.add(ServiceConstant.Header.X_AUTH_TENANT_ID,tenantId);
                                        });
                                    })
                                    .build();//创建一个新的exchange
                            //直接getHeaders()后调用add,set等修改方法会报错
                          //  exchange1.getRequest().getHeaders().add(ServiceConstant.Header.X_AUTH_TENANT_ID,tenantId);
                            //如果解析成功,说明我们的令牌没问题,放行过滤器
                            //我们顺便把租户的id保存到请求头中,然后放行到下游服务
                            return chain.filter(exchange1);
                        } catch (Exception e) {
                            e.printStackTrace();
                            try {
                                //拦截返回我们的自己的数据
                                return exchange.getResponse().writeAndFlushWith(Mono.just(Mono.just(
                                        DefaultDataBufferFactory.sharedInstance.allocateBuffer().write(objectMapper.writeValueAsString(R.fail("请检查网络")).getBytes(StandardCharsets.UTF_8)))));
                            } catch (JsonProcessingException ex) {
                                throw new RuntimeException(ex);
                            }

                        }
                    })
                    //当出现错误或者数据为空的时候,我们都校验失败 //拦截返回我们的自己的数据
                    .onErrorReturn(exchange.getResponse().writeAndFlushWith(Mono.just(Mono.just(
                            DefaultDataBufferFactory.sharedInstance.allocateBuffer().write(objectMapper.writeValueAsString(R.fail("请检查令牌")).getBytes(StandardCharsets.UTF_8))))))
                    .defaultIfEmpty(exchange.getResponse().writeAndFlushWith(Mono.just(Mono.just(DefaultDataBufferFactory.sharedInstance.allocateBuffer().write(objectMapper.writeValueAsString(R.fail("请检查令牌")).getBytes(StandardCharsets.UTF_8)))))).flatMap(result -> result);
        } catch (JsonProcessingException e) {
            //  throw new RuntimeException(e);
            try {
                return exchange.getResponse().writeAndFlushWith(Mono.just(Mono.just(DefaultDataBufferFactory.sharedInstance.allocateBuffer().write(objectMapper.writeValueAsString(R.fail("请检查令牌")).getBytes(StandardCharsets.UTF_8)))));
            } catch (JsonProcessingException ex) {
                throw new RuntimeException(ex);
            }
        }
    }

    @Override
    public int getOrder() {
        return 1;
    }
}
